list python hackerrank solution